while loop with python program